#ifndef _PRE_PROCESS_HPP
#define _PRE_PROCESS_HPP

#include <opencv2/core.hpp>
#include <opencv2/imgcodecs.hpp>
#include <opencv2/imgproc.hpp>

std::vector<cv::Mat> LoadCocoImages(std::string coco_path, int count = 10);

cv::Mat Preprocess(cv::Mat src_img, bool transpose = false, bool normlize = false, bool swapBR = true, int depth = CV_8U);

std::vector<cv::String> LoadImages(const std::string image_dir, const int batch_size);

#endif //_PRE_PROCESS_HPP
